DatabaseResourceTest.java

package com.collabinate.server.adminresources;

import static org.junit.Assert.*;

import java.io.StringReader;

import javax.xml.XMLConstants;
import javax.xml.transform.Source;
import javax.xml.transform.stream.StreamSource;
import javax.xml.validation.Schema;
import javax.xml.validation.SchemaFactory;
import javax.xml.validation.Validator;

import org.junit.Test;
import org.restlet.Response;
import org.restlet.data.MediaType;
import org.restlet.data.Status;

import com.collabinate.server.resources.GraphResourceTest;

/**
 * Tests for the Database Export Resource.
 * 
 * @author mafuba
 *
 */
public class DatabaseResourceTest extends GraphResourceTest
{

	@Test
	public void export_should_return_200()
	{
		assertEquals(Status.SUCCESS_OK, get().getStatus());
	}
	
	@Test
	public void export_should_have_xml_content_type()
	{
		assertEquals(MediaType.APPLICATION_XML,
				get().getEntity().getMediaType());
	}
	
	@Test
	public void export_should_contain_graphml()
	{
		assertTrue(isValidGraphml(get().getEntityAsText()));
	}
	
	@Test
	public void import_should_return_204()
	{
		Response response = 
				put(get().getEntityAsText(), MediaType.APPLICATION_XML);
		
		assertEquals(Status.SUCCESS_NO_CONTENT, response.getStatus());
	}
	
	public static boolean isValidGraphml(String graphMl)
	{
		SchemaFactory factory = SchemaFactory
				.newInstance(XMLConstants.W3C_XML_SCHEMA_NS_URI);
		Schema schema;
		try
		{
			schema = factory.newSchema(DatabaseResourceTest.class
					.getClassLoader().getResource(GRAPHML_SCHEMA_FILE));
		}
		catch (Exception e)
		{
			// TODO: fix offline validation
			e.printStackTrace();
			return true;
		}
		
		Validator validator = schema.newValidator();

		// create a source from a string
		Source source = new StreamSource(new StringReader(graphMl));

		// check input
		boolean isValid = true;
		try
		{
			validator.validate(source);
		}
		catch (Exception e)
		{
			e.printStackTrace();
			isValid = false;
		}

		return isValid;
	}
	
	@Override
	protected String getResourcePath()
	{
		return "/1/admin/database";
	}
	
	private static final String GRAPHML_SCHEMA_FILE = "graphml.xsd";
}